# coding:utf-8
import argparse

parser = argparse.ArgumentParser()


def add_argument_group(name):
    arg = parser.add_argument_group(name)
    return arg


misc_arg = add_argument_group('misc')
misc_arg.add_argument('--split', type=bool, default=True)
misc_arg.add_argument('--input_size', type=int, default=256,
                      help='multiplies of 256 by the structure of the model')
misc_arg.add_argument('--use_network', type=bool, default=False)

predict_arg = add_argument_group('predict')
predict_arg.add_argument('--num', type=int, default=None)
predict_arg.add_argument('--upload', type=bool, default=False)
predict_arg.add_argument('--sample_rate', type=int, default=None)
predict_arg.add_argument('--cinc_download', type=bool, default=False)


def get_config():
    config, unparsed = parser.parse_known_args()

    return config
